    # Step 1: Adjust `conditional_prompts` to match the sequence length of `txt_feats`
    if conditional_prompts.size(1) != sequence_length:
        # Use repeat to make conditional_prompts match the sequence length of txt_feats
        conditional_prompts = conditional_prompts.unsqueeze(1)  # Shape: [batch_size, 1, n_ctx, ctx_dim]
        conditional_prompts = conditional_prompts.repeat(1, sequence_length, 1, 1)  # Repeat across sequence dimension
        conditional_prompts = conditional_prompts.view(conditional_prompts.size(0), sequence_length, conditional_prompts.size(2))
